#pragma once

#include "../server.hpp"

#include <fstream>
#include <regex>
#include <sys/stat.h>

#define DEFALT_TIMEOUT 10

std::unordered_map<int, std::string> _statu_msg = {
        {100,  "Continue"},
        {101,  "Switching Protocol"},
        {102,  "Processing"},
        {103,  "Early Hints"},
        {200,  "OK"},
        {201,  "Created"},
        {202,  "Accepted"},
        {203,  "Non-Authoritative Information"},
        {204,  "No Content"},
        {205,  "Reset Content"},
        {206,  "Partial Content"},
        {207,  "Multi-Status"},
        {208,  "Already Reported"},
        {226,  "IM Used"},
        {300,  "Multiple Choice"},
        {301,  "Moved Permanently"},
        {302,  "Found"},
        {303,  "See Other"},
        {304,  "Not Modified"},
        {305,  "Use Proxy"},
        {306,  "unused"},
        {307,  "Temporary Redirect"},
        {308,  "Permanent Redirect"},
        {400,  "Bad Request"},
        {401,  "Unauthorized"},
        {402,  "Payment Required"},
        {403,  "Forbidden"},
        {404,  "Not Found"},
        {405,  "Method Not Allowed"},
        {406,  "Not Acceptable"},
        {407,  "Proxy Authentication Required"},
        {408,  "Request Timeout"},
        {409,  "Conflict"},
        {410,  "Gone"},
        {411,  "Length Required"},
        {412,  "Precondition Failed"},
        {413,  "Payload Too Large"},
        {414,  "URI Too Long"},
        {415,  "Unsupported Media Type"},
        {416,  "Range Not Satisfiable"},
        {417,  "Expectation Failed"},
        {418,  "I'm a teapot"},
        {421,  "Misdirected Request"},
        {422,  "Unprocessable Entity"},
        {423,  "Locked"},
        {424,  "Failed Dependency"},
        {425,  "Too Early"},
        {426,  "Upgrade Required"},
        {428,  "Precondition Required"},
        {429,  "Too Many Requests"},
        {431,  "Request Header Fields Too Large"},
        {451,  "Unavailable For Legal Reasons"},
        {501,  "Not Implemented"},
        {502,  "Bad Gateway"},
        {503,  "Service Unavailable"},
        {504,  "Gateway Timeout"},
        {505,  "HTTP Version Not Supported"},
        {506,  "Variant Also Negotiates"},
        {507,  "Insufficient Storage"},
        {508,  "Loop Detected"},
        {510,  "Not Extended"},
        {511,  "Network Authentication Required"}
};

// 文件后缀表
std::unordered_map<std::string, std::string> _mime_msg = {
        {".aac",        "audio/aac"},
        {".abw",        "application/x-abiword"},
        {".arc",        "application/x-freearc"},
        {".avi",        "video/x-msvideo"},
        {".azw",        "application/vnd.amazon.ebook"},
        {".bin",        "application/octet-stream"},
        {".bmp",        "image/bmp"},
        {".bz",         "application/x-bzip"},
        {".bz2",        "application/x-bzip2"},
        {".csh",        "application/x-csh"},
        {".css",        "text/css"},
        {".csv",        "text/csv"},
        {".doc",        "application/msword"},
        {".docx",       "application/vnd.openxmlformats-officedocument.wordprocessingml.document"},
        {".eot",        "application/vnd.ms-fontobject"},
        {".epub",       "application/epub+zip"},
        {".gif",        "image/gif"},
        {".htm",        "text/html"},
        {".html",       "text/html"},
        {".ico",        "image/vnd.microsoft.icon"},
        {".ics",        "text/calendar"},
        {".jar",        "application/java-archive"},
        {".jpeg",       "image/jpeg"},
        {".jpg",        "image/jpeg"},
        {".js",         "text/javascript"},
        {".json",       "application/json"},
        {".jsonld",     "application/ld+json"},
        {".mid",        "audio/midi"},
        {".midi",       "audio/x-midi"},
        {".mjs",        "text/javascript"},
        {".mp3",        "audio/mpeg"},
        {".mpeg",       "video/mpeg"},
        {".mpkg",       "application/vnd.apple.installer+xml"},
        {".odp",        "application/vnd.oasis.opendocument.presentation"},
        {".ods",        "application/vnd.oasis.opendocument.spreadsheet"},
        {".odt",        "application/vnd.oasis.opendocument.text"},
        {".oga",        "audio/ogg"},
        {".ogv",        "video/ogg"},
        {".ogx",        "application/ogg"},
        {".otf",        "font/otf"},
        {".png",        "image/png"},
        {".pdf",        "application/pdf"},
        {".ppt",        "application/vnd.ms-powerpoint"},
        {".pptx",       "application/vnd.openxmlformats-officedocument.presentationml.presentation"},
        {".rar",        "application/x-rar-compressed"},
        {".rtf",        "application/rtf"},
        {".sh",         "application/x-sh"},
        {".svg",        "image/svg+xml"},
        {".swf",        "application/x-shockwave-flash"},
        {".tar",        "application/x-tar"},
        {".tif",        "image/tiff"},
        {".tiff",       "image/tiff"},
        {".ttf",        "font/ttf"},
        {".txt",        "text/plain"},
        {".vsd",        "application/vnd.visio"},
        {".wav",        "audio/wav"},
        {".weba",       "audio/webm"},
        {".webm",       "video/webm"},
        {".webp",       "image/webp"},
        {".woff",       "font/woff"},
        {".woff2",      "font/woff2"},
        {".xhtml",      "application/xhtml+xml"},
        {".xls",        "application/vnd.ms-excel"},
        {".xlsx",       "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"},
        {".xml",        "application/xml"},
        {".xul",        "application/vnd.mozilla.xul+xml"},
        {".zip",        "application/zip"},
        {".3gp",        "video/3gpp"},
        {".3g2",        "video/3gpp2"},
        {".7z",         "application/x-7z-compressed"}
};


class Util
{
public:
    //分割
    static size_t Split(const std::string& src, const std::string& sep, std::vector<std::string>& ary)
    {
        size_t offset = 0;
        while(offset < src.size())
        {
            size_t pos = src.find(sep, offset);
            if(pos == offset)
            {
                offset = pos + sep.size();
                continue;
            }

            if(pos == std::string::npos)
            {
                ary.push_back(src.substr(offset));
                break;
            }
            else
            {
                ary.push_back(src.substr(offset, pos - offset));
                offset = pos + sep.size();
            }
        }
    }

    // 读取文件内容
    static bool ReadFile(const std::string& filename, std::string& buf)
    {
        std::ifstream ifs(filename, std::ios::binary);
        if(ifs.is_open() == 0)
        {
            LOG(Error, "open %s file failed", filename.c_str());
            return false;
        }

        size_t fsize = 0;
        ifs.seekg(0, ifs.end);
        fsize = ifs.tellg();
        ifs.seekg(0, ifs.beg);

        buf.resize(fsize);
        ifs.read(&buf[0], fsize);
        if(ifs.good() == 0)
        {
            LOG(Error, "read %s file failed", filename.c_str());
            ifs.close();
            return false;
        }

        ifs.close();
        return true;
    }
    // 向文件写入数据
    static bool WriteFile(const std::string& filename, const std::string& buf)
    {
        std::ofstream ofs(filename, std::ios::binary | std::ios::trunc);
        if(ofs.is_open() == 0)
        {
            LOG(Error, "open %s file failed", filename.c_str());
            return false;
        }
        ofs.write(buf.c_str(), buf.size());
        if(ofs.good() == 0)
        {
            LOG(Error, "write %s file failed", filename.c_str());
            ofs.close();
            return false;
        }

        ofs.close();
        return true;
    }

    // url编码
    // URL编码，避免URL中资源路径与查询字符串中的特殊字符与HTTP请求中特殊字符产生歧义
    // 编码格式：将特殊字符的ASCII值，转换为两个16进制字符，前缀%   C++ -> C%2B%2B
    // 不编码的特殊字符： RFC3986文档规定 . - _ ~ 字母，数字属于绝对不编码字符
    // RFC3986文档规定，编码格式 %HH 
    // W3C标准中规定，查询字符串中的空格，需要编码为+， 解码则是+转空格
    static std::string UrlEncode(const std::string& url, bool convert_space_to_plus)
    {
        std::string res;
        for(auto& c : url)
        {
            if(c == '.' || c == '-' || c == '_' || c == '~' || isalnum(c))
            {
                res += c;
                continue;
            }
            else if(c == ' ' && convert_space_to_plus == true)
            {
                res += '+';
                continue;
            }

            // 剩下的字符 编码格式 %HH
            char tmp[4] = { 0 };
            snprintf(tmp, 4, "%%%02X", c);
            res += tmp;
        }

        return res;
    }

    static char HexToI(char c)
    {
        if(c >= '0' && c <= '9')
            return c - '0';
        else if(c >= 'a' && c <= 'z')
            return c - 'a' + 10;
        else if(c >= 'A' && c <= 'Z')
            return c - 'A' + 10;
        else
            return -1;
    }

    // url解码
    static std::string UrlDecode(const std::string& url, bool convert_space_to_plus)
    {
        std::string res;
        for(int i = 0; i < url.size(); i++)
        {
            if(url[i] == '%' && i + 1 < url.size() && i + 2 < url.size())
            {
                char v1 = HexToI(url[i + 1]);
                char v2 = HexToI(url[i + 2]);
                char v = (v1 << 4) + v2;
                res += v;
                i += 2;
                continue;
            }
            if(url[i] == '+' && convert_space_to_plus == true)
            {
                res += ' ';
                continue;
            }

            res += url[i];
        }

        return res;
    }

    //响应状态码的描述信息获取
    static std::string StatuDesc(int statu)
    {
        
        if(_statu_msg.count(statu))
        {
            return _statu_msg[statu];
        }
        
        return "UnKown";
    }

    //根据文件后缀名获取文件mime
    static std::string ExtMine(const std::string& filename)
    {
        // a.txt
        size_t pos = filename.find_last_of('.');
        if(pos == std::string::npos)
        {
            return "application/octet-stream";
        }
        std::string ext = filename.substr(pos);
        if(_mime_msg.count(ext))
        {
            return _mime_msg[ext];
        }

        return "application/octet-stream"; 
    }

    // 是否为目录
    static bool IsDirectory(const std::string& filename)
    {
        struct stat st;
        // 获得文件属性
        int ret = stat(filename.c_str(), &st);
        if(ret < 0)
        {
            return false;
        }

        return S_ISDIR(st.st_mode);
    }

    // 是否为普通文件
    static bool IsRegular(const std::string& filename)
    {
        struct stat st;
        // 获得文件属性
        int ret = stat(filename.c_str(), &st);
        if(ret < 0)
        {
            return false;
        }

        return S_ISREG(st.st_mode);
    }

    //http请求的资源路径有效性判断
    // /index.html  --- 前边的/叫做相对根目录  映射的是某个服务器上的子目录
    // 想表达的意思就是，客户端只能请求相对根目录中的资源，其他地方的资源都不予理会
    // /../login, 这个路径中的..会让路径的查找跑到相对根目录之外，这是不合理的，不安全的
    static bool IsValidPath(const std::string &path)
    {
        std::vector<std::string> subdir;
        Split(path, "/", subdir);
        int level = 0;
        for(auto& dir : subdir)
        {
            if(dir == "..")
            {
                level--;
                // 当level小于0的时候，相当于走出了相对目录，有问题
                if(level < 0)
                    return false;
            }
            else
            {
                level++;
            }
        }

        return true;
    }
};


class HttpRequest
{
public:
    HttpRequest()
        :_version("HTTP/1.1") 
    {}

    void SetHeader(const std::string& key, const std::string& val)
    {
        _headers.insert({ key, val });
    }

    bool HasHeader(const std::string& key)
    {
        if(_headers.count(key))
            return true;
        else
            return false;
    }

    std::string GetHeader(const std::string& key)
    {
        if(_headers.count(key))
            return _headers[key];
        else
            return "";
    }

    void SetParam(const std::string& key, const std::string& val)
    {
        _params.insert({ key, val });
    }

    bool HasParam(const std::string& key)
    {
        if(_params.count(key))
            return true;
        else
            return false;
    }

    std::string GetParam(const std::string& key)
    {
        if(_params.count(key))
            return _params[key];
        else
            return "";
    }

    // 获取正文长度
    size_t ContentLength()
    {
        // Content-Length: xxxx\r\n
        if(_headers.count("Content-Length") == 0)
        {
            return 0;
        }
        std::string len = _headers["Content-Length"];
        return std::stol(len);
    }

    // 判断是否是短链接
    bool Close() 
    {
        // 短链接 : 没有Connection字段，或者有Connection但是值是close，则都是，
        // 长连接 : 其他
        if(_headers.count("Connection") && _headers["Connection"] == "keep-alive")
            return false;
        else
            return true;
    }

    void ReSet() 
    {
        _method.clear();
        _path.clear();
        _version = "HTTP/1.1";
        _body.clear();
        std::smatch match;
        _matches.swap(match);
        _headers.clear();
        _params.clear();
    }

public:
    std::string _method;    // 请求方法
    std::string _path;      // 资源路径
    std::string _version;   // 协议版本
    std::string _body;      // 请求正文
    std::smatch _matches;   // 资源路径的正则提取数据
    std::unordered_map<std::string, std::string> _headers; // 头部字段
    std::unordered_map<std::string, std::string> _params;  // 查询字符串
};


class HttpResponse 
{
public:
    HttpResponse()
        :_redirect_flag(false)
        , _statu(200) 
    {}

    HttpResponse(int statu)
        :_redirect_flag(false)
        , _statu(statu) 
    {} 

    //插入头部字段
    void SetHeader(const std::string& key, const std::string& val)
    {
        _headers.insert({ key, val });
    }

    bool HasHeader(const std::string& key)
    {
        if(_headers.count(key))
            return true;
        else
            return false;
    }
    
    // 获取头部字段
    std::string GetHeader(const std::string& key)
    {
        if(_headers.count(key))
            return _headers[key];
        else
            return "";
    }

    // 设置正文
    void SetContent(const std::string& body,  const std::string& type = "text/html") 
    {
        _body = body;
        SetHeader("Content-Type", type);
    }

    // 设置重定向  --默认是302临时重定向
    void SetRedirect(const std::string& url, int statu = 302) 
    {
        _statu = statu;
        _redirect_flag = true;
        _redirect_url = url;
    }

    //判断是否是短链接
    bool Close()
    {
        // 短链接 : 没有Connection字段，或者有Connection但是值是close，则都是，
        // 长连接 : 其他
        if(_headers.count("Connection") && _headers["Connection"] == "keep-alive")
            return false;
        else
            return true;
    }

    void ReSet() 
    {
        _statu = 200;
        _redirect_flag = false;
        _body.clear();
        _redirect_url.clear();
        _headers.clear();
    }

public:
    int _statu;
    std::string _body;          // 正文
    bool _redirect_flag;        // 重定向标志
    std::string _redirect_url;  // 重定向urrl 
    std::unordered_map<std::string, std::string> _headers;
};


typedef enum 
{
    RECV_HTTP_ERROR,  // 接收错误
    RECV_HTTP_LINE,   // 接收请求行： 请求方法 url 协议类型
    RECV_HTTP_HEAD,   // 接收头部字段 key=value\r\n
    RECV_HTTP_BODY,   // 接收正文
    RECV_HTTP_OVER    // 接受完毕
}HttpRecvStatu;


#define MAX_LINE 8192

class HttpContext
{
private:
    // 解析请求行
    bool ParseHttpLine(const std::string& line)
    {
        std::smatch matches;
        std::regex e("(GET|HEAD|POST|PUT|DELETE) ([^?]*)(?:\\?(.*))? (HTTP/1\\.[01])(?:\n|\r\n)?", std::regex::icase);
        bool ret = std::regex_match(line, matches, e);
        if(ret == false)
        {
            _recv_statu = RECV_HTTP_ERROR;
            _resp_statu = 400;//BAD REQUEST
            return false;
        }

        // 请求方法的获取-将其全部变为大写
        _request._method = matches[1];
        std::transform(_request._method.begin(), _request._method.end(), _request._method.begin(), ::toupper);

        //资源路径的获取，需要进行URL解码操作，但是不需要+转空格
        _request._path = Util::UrlDecode(matches[2], false);

        //协议版本的获取
        _request._version = matches[4];

        //查询字符串的获取与处理
        std::vector<std::string> query_string_arry;
        std::string query_string = matches[3];

        // 查询字符串的格式 key=val&key=val....., 先以 & 符号进行分割，得到各个字串
        Util::Split(query_string, "&", query_string_arry);
        for(auto& str : query_string_arry)
        {
            // 针对各个字串，以 = 符号进行分割
            // 得到key 和val， 得到之后也需要进行URL解码
            size_t pos = str.find("=");
            if(pos == std::string::npos)
            {
                _recv_statu = RECV_HTTP_ERROR;
                _resp_statu = 400;//BAD REQUEST
                return false;
            }
            std::string key = Util::UrlDecode(str.substr(0, pos), true);
            std::string val = Util::UrlDecode(str.substr(pos + 1), true);
            _request.SetParam(key, val);
        }

        return true;
    }


    // 接收请求行
    bool RecvHttpLine(Buffer* buf)
    {
        if (_recv_statu != RECV_HTTP_LINE) 
            return false;
        
        // 1，获取一行数据，末尾有换行符
        std::string line = buf->GetLineAndPop();
        // 2. 需要考虑的一些要素：缓冲区中的数据不足一行， 获取的一行数据超大
        if(line.size() == 0)
        {
            // 缓冲区中的数据不足一行，则需要判断缓冲区的可读数据长度，
            // 如果很长了都不足一行，这是有问题的
            if(buf->ReadableSize() > MAX_LINE)
            {
                _recv_statu = RECV_HTTP_ERROR;
                _resp_statu = 414;  // URL TOO LING
                return false;
            }

            // 缓冲区中数据不足一行，但是也不多，就等等新数据的到来
            return true;
        }

        if(line.size() > MAX_LINE)
        {
            _recv_statu = RECV_HTTP_ERROR;
            _resp_statu = 414;  // URL TOO LING
            return false;
        }

        if(ParseHttpLine(line) == false)
        {
            return false;
        }
        //首行处理完毕，进入头部获取阶段
        _recv_statu = RECV_HTTP_HEAD;

        return true;    
    }

    // 解析头部
    bool ParseHttpHead(std::string& line)
    {
        // 头部字段 key: val\r\n
        //末尾是回车换行则去掉换行字符
        if (line.back() == '\n') 
            line.pop_back();
        if (line.back() == '\r') 
            line.pop_back();

        size_t pos = line.find(": ");
        if(pos == std::string::npos)
        {
            _recv_statu = RECV_HTTP_ERROR;
            _resp_statu = 400;//BAD REQUEST
            return false;
        }
        std::string key = line.substr(0, pos);
        std::string val = line.substr(pos + 2);
        
        _request.SetHeader(key, val);

        return true;
    }

    // 接收头部
    bool RecvHttpHead(Buffer* buf)
    {
        if(_recv_statu != RECV_HTTP_HEAD)
            return false;

        while(1)
        {
            // 1，获取一行数据，末尾有换行符
            std::string line = buf->GetLineAndPop();
            // 2. 需要考虑的一些要素：缓冲区中的数据不足一行， 获取的一行数据超大
            if(line.size() == 0)
            {
                // 缓冲区中的数据不足一行，则需要判断缓冲区的可读数据长度，
                // 如果很长了都不足一行，这是有问题的
                if(buf->ReadableSize() > MAX_LINE)
                {
                    _recv_statu = RECV_HTTP_ERROR;
                    _resp_statu = 414;  // URL TOO LING
                    return false;
                }

                // 缓冲区中数据不足一行，但是也不多，就等等新数据的到来
                return true;
            }

            if(line.size() > MAX_LINE)
            {
                _recv_statu = RECV_HTTP_ERROR;
                _resp_statu = 414;  // URL TOO LING
                return false;
            }

            // 读到空行break
            if(line == "\n" || line == "\r\n")
            {
                break;
            }            

            if(ParseHttpHead(line) == false)
            {
                return false;
            }
        }

        //头部处理完毕，进入正文获取阶段
        _recv_statu = RECV_HTTP_BODY;
        

        return true;
    }

    // 接收正文
    bool RecvHttpBody(Buffer* buf)
    {    
        if(_recv_statu != RECV_HTTP_BODY)
            return false;
        
        // 1， 获取正文长度
        size_t context_len = _request.ContentLength();
        if(context_len == 0)
        {
            // 没有正文，请求解析完毕
            _recv_statu = RECV_HTTP_OVER;
            return true;
        }
        
        // 当前接收了一部分正文，但是没有接受完
        // 2， 获取实际还需接收的数据
        size_t real_len = context_len - _request._body.size();

        // 3，接收正文放body中， 但是要判断是不是全部的正文
        // 3-1, 缓冲区包含完整的正文
        if(buf->ReadableSize() >= real_len)
        {
            _request._body.append(buf->ReadPos(), real_len); //传参传的不对呀，应该传地址，懂了
            buf->MoveReadOffset(real_len);
            // std::cout << "接收的正文是：" << _request._body << std::endl;
            _recv_statu = RECV_HTTP_OVER;
            return true;
        }
        // 3-2, 缓冲区的正文不完整
        _request._body.append(buf->ReadPos(), buf->ReadableSize());
        buf->MoveReadOffset(buf->ReadableSize());

        return true; 
    }

public:

    HttpContext()
        : _resp_statu(200)
        , _recv_statu(RECV_HTTP_LINE)
    {}

    // 返回响应状态码
    int GetRespStatu()
    {
        return _resp_statu;
    }

    // 返回接收解析状态
    HttpRecvStatu GetRecvStatu()
    {
        return _recv_statu;
    }

    // 返回解析后的信息
    HttpRequest& GetRequest()
    {
        return _request;
    }

    void RecvHttpRequest(Buffer *buf) 
    {
        //不同的状态，做不同的事情，但是这里不要break， 因为处理完请求行后，应该立即处理头部，而不是退出等新数据
        switch(_recv_statu) 
        {
            case RECV_HTTP_LINE: RecvHttpLine(buf);
            case RECV_HTTP_HEAD: RecvHttpHead(buf);
            case RECV_HTTP_BODY: RecvHttpBody(buf);
        }

        return;
    }

    void ReSet() 
    {
        _resp_statu = 200;
        _recv_statu = RECV_HTTP_LINE;
        _request.ReSet();
    }

private:
    int _resp_statu;            //响应状态码
    HttpRecvStatu _recv_statu;  //当前接收及解析的阶段状态
    HttpRequest _request;       //已经解析得到的请求信息
};



class HttpServer
{
private:
    using Handler = std::function<void(const HttpRequest&, HttpResponse*)>;
    using Handlers = std::vector<std::pair<std::regex, Handler>>;
    // 设置上下文
    void OnConnected(const PtrConnection& conn)
    {
        conn->SetContext(HttpContext());
        LOG(Info, "NEW CONNECTION %p", conn.get()); // get 返回裸指针
    }
    
    bool IsFileHandler(const HttpRequest& req)
    {
        // 1, 必须设置了静态资源根目录
        if(_basedir.size() == 0)
        {
            return false;
        } 
        // 2, 请求方法必须是 GET HEAD
        if(req._method != "GET" && req._method != "HEAD")
        {
            return false;
        }
        // 3, 请求资源路径合法
        if(Util::IsValidPath(req._path) == false)
        {   
            return false;
        }
        // 4, 请求资源要是一个普通文件
        std::string req_path = _basedir + req._path;
        if(req_path.back() == '/')
        {
            req_path += "index.html";
        }
        if(Util::IsRegular(req_path) == false)
        {
            return false;
        }

        return true;
    }

    // 静态资源的请求处理
    void FileHandler(const HttpRequest& req, HttpResponse* rsp)
    {
        std::string req_path = _basedir + req._path;
        if(req_path.back() == '/')
        {
            req_path += "index.html";
        }
        // 将静态资源文件的数据读取出来，放到rsp的_body中, 并设置mime
        bool ret = Util::ReadFile(req_path, (rsp->_body));
        if(ret == false)
        {
            return;
        }
        std::string mime = Util::ExtMine(req_path);
        rsp->SetHeader("Content-Type", mime);
    }
    // 功能性请求处理
    void Dispatcher(HttpRequest& req, HttpResponse* rsp, const Handlers& handlers)
    {
        //在对应请求方法的路由表中，查找是否含有对应资源请求的处理函数，有则调用，没有则发挥404
        for(auto& handler : handlers)
        {
            const std::regex& re = handler.first;
            const Handler& functor = handler.second;
            bool ret = std::regex_match(req._path, req._matches, re);
            if(ret == true)
            {
                functor(req, rsp);
                return;
            }      
        }
        rsp->_statu = 404; // NOT fund
    }

    // 请求路由 业务处理
    void Route(HttpRequest& req, HttpResponse* rsp)
    {
        // std::cout << req._method << std::endl;
        if(IsFileHandler(req) == true)
        {
            FileHandler(req, rsp);
            return;
        }

        if(req._method == "GET" || req._method == "HEAD")
        {
            LOG(Info, "收到GET请求");
            Dispatcher(req, rsp, _get_route);
            return;
        }
        else if(req._method == "POST")
        {
            LOG(Info, "收到POST请求");
            Dispatcher(req, rsp, _post_route);
            return;
        }
        else if(req._method == "PUT")
        {
            LOG(Info, "收到PUT请求");
            Dispatcher(req, rsp, _put_route);
            return;
        }
        else if(req._method == "DELETE")
        {
            Dispatcher(req, rsp, _delete_route);
            return;
        }

        // Method Not Allowed
        rsp->_statu = 404;
        return;
    }
    

    // 将HttpResponse中的要素按照http协议格式进行组织，发送
    void WriteResponse(const PtrConnection& conn, HttpRequest& req, HttpResponse& rsp)
    {
        // 1，先完善 头部字段
        // 长短链接
        if(req.Close() == true)
        {
            // LOG(Debug, "close");
            rsp.SetHeader("Connection", "close");
        }
        else
        {
            // LOG(Debug, "keel-alive");
            rsp.SetHeader("Connection", "keep-alive");
        }
        // 正文长度
        if(rsp._body.empty() == 0 && rsp.HasHeader("Content-Length") == false)
        {
            rsp.SetHeader("Content-Length", std::to_string(rsp._body.size()));
        }
        // 正文类型
        if(rsp._body.empty() == 0 && rsp.HasHeader("Content-Type") == false)
        {
            rsp.SetHeader("Content-Type", "application/octet-stream");
        }

        // 2, 将rsp中的要素按照HTTP协议格式进行组织
        std::stringstream rsp_str;
        // 状态行： HTTP版本 状态码 状态信息\r\n
        rsp_str << req._version << " " << std::to_string(rsp._statu) << " " << Util::StatuDesc(rsp._statu) << "\r\n";
        // 响应头部字段 key: val\r\n
        for(auto& ch : rsp._headers)
        {
            rsp_str << ch.first << ": " << ch.second << "\r\n";
        }
        rsp_str << "\r\n";     // 空行
        rsp_str << rsp._body;  // 正文

        // std::cout << "发送的数据是： " << rsp_str.str().c_str() << std::endl;
        // 3, 发送数据
        conn->Send(rsp_str.str().c_str(), rsp_str.str().size());
    }

    // 错误处理
    void ErrorHandler(const HttpRequest &req, HttpResponse *rsp)
    {
        // 1， 组织一个错误响应页面
        std::string body;
        body += "<html>";
        body += "<head>";
        body += "<meta http-equiv=Content-Type content='text/html; charset=utf-8'>";
        body += "</head>";
        body += "<h1>";
        body += std::to_string(rsp->_statu);
        body += " ";
        body += Util::StatuDesc(rsp->_statu);
        body += "</h1>";
        body += "</body>";
        body += "</html>";

        // 将页面数据当做响应正文放入rsp中
        rsp->SetContent(body, "text/html");
    }

    // 缓冲区数据接收 处理
    void OnMessage(const PtrConnection& conn, Buffer* buf)
    {
        while(buf->ReadableSize() > 0)
        {
            // 1, 获取上下文 
            HttpContext* context = conn->GetContext()->GetVal<HttpContext>();
            // 2, 通过上下文对缓冲区的数据进行解析 得到HttpRequest对象
            
            context->RecvHttpRequest(buf);

            HttpResponse rsp(context->GetRespStatu());
            HttpRequest& req = context->GetRequest();

            //如果缓冲区的数据解析出错，就直接回复出错响应
            if(context->GetRespStatu() >= 400)
            {
                // 进行错误响应 关闭链接
                ErrorHandler(req, &rsp);
                WriteResponse(conn, req, rsp);
                context->ReSet();
                //出错了就把缓冲区数据清空
                buf->MoveReadOffset(buf->ReadableSize());
                conn->Shutdown();//关闭连接
                return;
            }

            //当前请求还没有接收完整,则退出，等新数据到来再重新继续处理
            if (context->GetRecvStatu() != RECV_HTTP_OVER) 
            {
                return;
            }

            // 3, 请求路由 业务处理
            Route(req, &rsp);
            // 4, 对HttpResponse进行组织发送
            WriteResponse(conn, req, rsp);
            // 5, 重置上下文
            context->ReSet();
            // 6, 根据长短链接判断是关闭还是继续处理
            if (rsp.Close() == true) 
            {
                //短链接则直接关闭
                conn->Shutdown();
            }
        }
    }
public:
    HttpServer(int port, int timeout = DEFALT_TIMEOUT)
        : _server(port)
    {
        _server.EnableInactiveRelease(timeout);
        _server.SetConnectCallback(std::bind(&HttpServer::OnConnected, this, std::placeholders::_1));
        _server.SetMessageCallback(std::bind(&HttpServer::OnMessage, this, std::placeholders::_1, std::placeholders::_2));
    }

    void Get(const std::string& pattern, const Handler& handler)
    {
        _get_route.push_back(std::make_pair(std::regex(pattern), handler));
    }
    void Post(const std::string& pattern, const Handler& handler)
    {
        _post_route.push_back(std::make_pair(std::regex(pattern), handler));
    }
    void Put(const std::string& pattern, const Handler& handler)
    {
        _put_route.push_back(std::make_pair(std::regex(pattern), handler));
    }
    void Delete(const std::string& pattern, const Handler& handler)
    {
        _delete_route.push_back(std::make_pair(std::regex(pattern), handler));
    }
    void SetBaseDir(const std::string& path)
    {
        assert(Util::IsDirectory(path) == true);
        _basedir = path;
    }
    void SetThreadCount(int count)
    {
        _server.SetThreadCount(count);
    }
    void Listen()
    {
        _server.Start();
    }


private:
    Handlers _get_route;
    Handlers _post_route;
    Handlers _put_route;
    Handlers _delete_route;
    std::string _basedir;  // 静态资源目录
    TcpServer _server;
};